from env_tools import get_env_value
from langchain_deepseek import ChatDeepSeek

# 获取apikey
api_key = get_env_value("DEEPSEEK_API_KEY")
if api_key is None:
    raise ValueError("请设置环境变量DEEPSEEK_API_KEY")

# 初始化deepseek
model = ChatDeepSeek(
    model = "deepseek-chat",            # 指定使用的模型
    temperature = 0,                    # 采样温度，指定0表示输出更确定性，取值在[0,1]之间，越小越精确，通常设置为0.7
    max_tokens=None,                    # 最大输出长度，None表示无限长度
    timeout=None,                       # 请求超时时间，None表示无限超时
    max_retries=2,                      # 最大重试次数
    api_key=api_key
)

print(model.invoke("1+1是多少？"))